from runner import Runner
from common.arguments import get_args
from common.utils import make_uav_env



if __name__ == '__main__':  
    # get the params
    args = get_args()                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                   
    env, args = make_uav_env(args)
    #plot_environment(env)
    runner = Runner(args, env) 
    if args.uavtrain:
        runner.uav_train1()
    elif args.evaluate:
        runner.uav_eval()




